<?php
/**
 * @title 446. 等差数列划分 II - 子序列
 * @author start2004
 */

// ini_set("memory_limit", "128M");

// class Solution {
//
//     function find() {
//
//     }
// }

class Solution {

    /**
     * @param Integer[] $nums
     * @return Integer
     */
    function numberOfArithmeticSlices($nums) {
        /**
         * @since 2021-08-11 长度
         */
        $len = count($nums);

        /**
         * @since 2021-08-11 记录每个数字位置
         */
        $posArray = [];
        $cntArray = [];
        for ($i=0; $i<$len; $i++){
            $num = $nums[$i];
            $posArray[$num][] = $i;

            /**
             * @since 2021-08-12 计数
             */
            if(isset($cntArray[$num])){
                $cntArray[$num] += 1;
            } else {
                $cntArray[$num] = 1;
            }
        }
        // return $posArray;

        /**
         * @since 2021-08-11 相同元素最多数量
         */
        $maxCnt = max($cntArray);

        /**
         * @since 2021-08-11 统计
         */
        $dpArray = [3 => 1];
        for($i=4; $i<=$maxCnt; $i++){
            $a = ($i-1)*($i-2);
            $b = 2;
            $cnt = $dpArray[$i-1] + $a/$b;

            for($j=3; $j<=$i-1; $j++){
                $a = $a*($i-$j);
                $b = $b*$j;
                $cnt += $a/$b;
            }
            $dpArray[$i] = $cnt;
        }
        // return $dpArray;

        /**
         * @since 2021-08-11 结果集
         */
        $result = 0;

        /**
         * @since 2021-08-11 等差d=0
         */
        foreach ($posArray as $pArray){
            $cnt = count($pArray);
            if($cnt>=3){
                $result += $dpArray[$cnt];
            } else {}
        }

        /**
         * @return
         */
        return $result;
    }
}

/**
 * @url http://127.0.0.1/leetcode/202108/2021.08.11.php
 */
$datas = [
    // [[2,4,6,8,10]],  // 7
    [[7,7,7,7,7]], // 16
];

include_once dirname(__DIR__) . DIRECTORY_SEPARATOR ."xhprof.php";
$xhprof = new Xhprof();
foreach ($datas as $data){
    if(count($data) <= 20){
        var_dump($data);
    } else {
        var_dump(count($data));
    }

    $obj = new Solution();
    $result = $obj->numberOfArithmeticSlices(...$data);
    // $result = $obj->($xhprof->tree($data));
    // $result = $obj->($xhprof->listNode($data));
    // $result = $obj->find(...$data);
    var_dump($result);
    // if(count($result)<=20){
    //     var_dump($result);
    // } else {
    //     var_dump(count($result));
    // }
    echo str_repeat(PHP_EOL, 3);
    echo str_repeat("<br>", 3);
}

// foreach ($datas as $data){
//     $obj = new $data[0][0](...$data[1][0]);
//
//     for ($i=1; $i<count($data[0]); $i++){
//         $func = $data[0][$i];
//         $param = $data[1][$i];
//         $result = $obj->$func(...$param);
//
//         echo $data[0][$i] ."(\"". implode(",", $data[1][$i]) ."\") ";
//             if($result === true){
//                 echo "True";
//             } elseif($result === false) {
//                 echo "False";
//             } elseif($result === null) {
//                 echo "Null";
//             } elseif(is_array($result)) {
//                 var_dump($result);
//             } else {
//                 echo $result;
//             }
//         echo PHP_EOL;
//     }
//
//     echo str_repeat(PHP_EOL, 3);
// }
$xhprof->end();
